tensor([[[[[0.1500, 0.1100],
           [0.1100, 0.1500]],

          [[0.1800, 0.0500],
           [0.0500, 0.1800]]],


         [[[0.1800, 0.0500],
           [0.0500, 0.1800]],

          [[0.9900, 1.0000],
           [0.0100, 0.0100]]]],



        [[[[0.2800, 0.2800],
           [0.2800, 0.2800]],

          [[0.2400, 0.2600],
           [0.2600, 0.2400]]],


         [[[0.2400, 0.2600],
           [0.2600, 0.2400]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.5100, 0.5400],
           [0.5400, 0.5100]],

          [[0.5200, 0.5600],
           [0.5600, 0.5200]]],


         [[[0.5200, 0.5600],
           [0.5600, 0.5200]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.0600, 0.0700],
           [0.0700, 0.0600]],

          [[0.0600, 0.1200],
           [0.1200, 0.0600]]],


         [[[0.0600, 0.1200],
           [0.1200, 0.0600]],

          [[0.0100, 0.0000],
           [0.9900, 0.9900]]]]], grad_fn=<RoundBackward1>)